{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "module.decoder.decoder.22.bn_list.1.weight\n",
      "module.decoder.decoder.15.bn_list.1.bias\n",
      "module.decoder.decoder.2.bn_list.0.weight\n",
      "module.encoder.lin1.bias\n",
      "module.decoder.decoder.2.bn1.bias\n",
      "module.decoder.decoder.10.conv_list.0.weight\n",
      "module.decoder.decoder.11.bn_list.0.bias\n",
      "module.decoder.decoder.9.bn_list.0.bias\n",
      "module.decoder.decoder.11.bn1.num_batches_tracked\n",
      "module.decoder.decoder.11.bn1.running_mean\n",
      "module.decoder.decoder.21.bn1.weight\n",
      "module.decoder.decoder.17.bn2.weight\n",
      "module.decoder.decoder.22.conv_list.0.weight\n",
      "module.decoder.decoder.3.bn_list.1.bias\n",
      "module.decoder.decoder.10.bn_list.0.bias\n",
      "module.decoder.decoder.18.bn_list.1.running_var\n",
      "module.decoder.decoder.23.conv_list.1.weight\n",
      "module.decoder.decoder.13.bn_list.1.weight\n",
      "module.decoder.decoder.1.bn_list.0.bias\n",
      "module.decoder.decoder.8.last_conv.bias\n",
      "module.decoder.decoder.14.bn_list.1.running_var\n",
      "module.decoder.decoder.7.bn_list.1.bias\n",
      "module.decoder.decoder.11.conv1.bias\n",
      "module.decoder.decoder.12.bn_list.0.running_var\n",
      "module.decoder.decoder.19.conv_list.0.weight\n",
      "module.decoder.decoder.20.conv2.weight\n",
      "module.decoder.decoder.16.bn_list.1.bias\n",
      "module.decoder.decoder.1.bn_list.0.weight\n",
      "module.decoder.decoder.11.bn2.weight\n",
      "module.decoder.decoder.10.last_conv.weight\n",
      "module.decoder.decoder.9.conv2.weight\n",
      "module.decoder.decoder.4.bn_list.0.bias\n",
      "module.decoder.decoder.9.bn_list.0.running_mean\n",
      "module.encoder.conv2.weight\n",
      "module.decoder.decoder.23.conv1.weight\n",
      "module.decoder.decoder.9.conv1.bias\n",
      "module.decoder.decoder.6.bn1.weight\n",
      "module.decoder.decoder.7.bn2.running_var\n",
      "module.decoder.decoder.2.conv2.bias\n",
      "module.decoder.decoder.12.bn2.num_batches_tracked\n",
      "module.decoder.decoder.18.bn1.weight\n",
      "module.decoder.decoder.16.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.17.bn1.running_var\n",
      "module.decoder.decoder.23.last_conv.bias\n",
      "module.decoder.decoder.9.bn1.weight\n",
      "module.decoder.decoder.7.bn_list.0.running_var\n",
      "module.encoder.bn5.weight\n",
      "module.decoder.decoder.5.bn_list.1.running_mean\n",
      "module.decoder.decoder.7.bn1.weight\n",
      "module.decoder.decoder.17.bn1.weight\n",
      "module.decoder.decoder.21.conv2.bias\n",
      "module.decoder.decoder.11.last_conv.bias\n",
      "module.decoder.decoder.12.bn1.bias\n",
      "module.decoder.decoder.22.bn_list.1.running_var\n",
      "module.decoder.decoder.4.conv_list.0.weight\n",
      "module.decoder.decoder.20.conv_list.0.weight\n",
      "module.decoder.decoder.6.bn_list.1.weight\n",
      "module.decoder.decoder.2.conv2.weight\n",
      "module.decoder.decoder.18.bn_list.0.running_mean\n",
      "module.decoder.decoder.6.last_conv.bias\n",
      "module.decoder.decoder.18.conv1.weight\n",
      "module.decoder.decoder.11.conv_list.0.weight\n",
      "module.decoder.decoder.21.bn2.running_mean\n",
      "module.decoder.decoder.14.bn1.num_batches_tracked\n",
      "module.decoder.decoder.20.conv1.weight\n",
      "module.decoder.decoder.16.bn_list.1.weight\n",
      "module.decoder.decoder.22.bn1.running_var\n",
      "module.decoder.decoder.24.conv_list.0.weight\n",
      "module.decoder.decoder.13.bn1.num_batches_tracked\n",
      "module.decoder.decoder.13.bn_list.1.running_var\n",
      "module.decoder.decoder.15.bn1.weight\n",
      "module.decoder.decoder.10.bn2.running_var\n",
      "module.decoder.decoder.24.bn_list.1.running_mean\n",
      "module.decoder.decoder.12.bn_list.1.bias\n",
      "module.decoder.decoder.11.bn_list.1.running_var\n",
      "module.decoder.decoder.21.last_conv.bias\n",
      "module.decoder.decoder.11.bn_list.0.running_mean\n",
      "module.decoder.decoder.19.conv2.bias\n",
      "module.decoder.decoder.4.bn1.bias\n",
      "module.decoder.decoder.18.conv1.bias\n",
      "module.encoder.bn1.bias\n",
      "module.decoder.decoder.8.bn1.running_mean\n",
      "module.decoder.decoder.3.bn_list.1.weight\n",
      "module.decoder.decoder.5.conv2.weight\n",
      "module.decoder.decoder.2.conv_list.1.weight\n",
      "module.decoder.decoder.8.conv2.weight\n",
      "module.decoder.decoder.1.bn1.running_mean\n",
      "module.decoder.decoder.0.conv_list.0.weight\n",
      "module.decoder.decoder.21.bn_list.1.bias\n",
      "module.decoder.decoder.2.bn2.weight\n",
      "module.decoder.decoder.7.bn2.num_batches_tracked\n",
      "module.decoder.decoder.20.bn_list.0.bias\n",
      "module.decoder.decoder.6.bn2.bias\n",
      "module.decoder.decoder.24.bn1.running_var\n",
      "module.decoder.decoder.1.last_conv.bias\n",
      "module.decoder.decoder.10.bn2.weight\n",
      "module.decoder.decoder.23.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.3.conv2.bias\n",
      "module.decoder.decoder.6.conv2.weight\n",
      "module.decoder.decoder.3.last_conv.weight\n",
      "module.decoder.decoder.3.bn2.running_var\n",
      "module.decoder.decoder.13.conv1.weight\n",
      "module.decoder.decoder.24.bn_list.0.bias\n",
      "module.encoder.conv3.bias\n",
      "module.decoder.decoder.7.bn1.num_batches_tracked\n",
      "module.decoder.decoder.8.bn_list.1.running_mean\n",
      "module.decoder.decoder.20.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.22.conv2.weight\n",
      "module.decoder.decoder.22.bn_list.1.running_mean\n",
      "module.decoder.decoder.11.bn_list.0.running_var\n",
      "module.decoder.decoder.5.bn_list.0.running_mean\n",
      "module.decoder.decoder.24.conv_list.1.weight\n",
      "module.decoder.decoder.1.conv_list.0.weight\n",
      "module.decoder.decoder.4.bn_list.0.weight\n",
      "module.decoder.decoder.12.conv_list.0.weight\n",
      "module.decoder.decoder.6.conv_list.1.bias\n",
      "module.decoder.decoder.8.bn_list.1.bias\n",
      "module.decoder.decoder.19.conv_list.0.bias\n",
      "module.decoder.decoder.6.bn_list.0.running_var\n",
      "module.decoder.decoder.15.conv2.bias\n",
      "module.decoder.decoder.15.bn_list.0.bias\n",
      "module.decoder.decoder.15.conv2.weight\n",
      "module.decoder.decoder.15.bn1.bias\n",
      "module.decoder.decoder.5.bn2.running_mean\n",
      "module.decoder.decoder.23.bn1.num_batches_tracked\n",
      "module.decoder.decoder.16.last_conv.bias\n",
      "module.encoder.bn4.running_var\n",
      "module.decoder.decoder.21.bn2.running_var\n",
      "module.decoder.decoder.3.bn1.running_mean\n",
      "module.decoder.decoder.8.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.23.bn1.bias\n",
      "module.decoder.decoder.24.bn2.num_batches_tracked\n",
      "module.decoder.decoder.23.bn_list.1.running_mean\n",
      "module.decoder.decoder.21.bn_list.0.bias\n",
      "module.decoder.decoder.12.last_conv.bias\n",
      "module.decoder.decoder.4.bn_list.1.running_mean\n",
      "module.decoder.decoder.2.bn2.running_var\n",
      "module.decoder.decoder.11.conv1.weight\n",
      "module.decoder.decoder.8.bn1.running_var\n",
      "module.decoder.decoder.17.conv_list.1.bias\n",
      "module.decoder.decoder.1.conv2.bias\n",
      "module.decoder.decoder.8.bn2.num_batches_tracked\n",
      "module.decoder.decoder.3.bn2.running_mean\n",
      "module.decoder.decoder.14.conv_list.0.weight\n",
      "module.decoder.decoder.18.bn_list.1.bias\n",
      "module.decoder.decoder.13.bn1.running_var\n",
      "module.decoder.decoder.18.bn2.bias\n",
      "module.decoder.decoder.22.bn1.num_batches_tracked\n",
      "module.decoder.decoder.12.bn2.running_var\n",
      "module.decoder.decoder.24.bn_list.1.running_var\n",
      "module.encoder.conv1.bias\n",
      "module.decoder.decoder.16.bn_list.0.bias\n",
      "module.decoder.decoder.2.bn1.weight\n",
      "module.decoder.decoder.19.bn1.running_var\n",
      "module.decoder.decoder.2.last_conv.bias\n",
      "module.decoder.decoder.12.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.2.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.6.bn1.bias\n",
      "module.decoder.decoder.9.bn2.running_var\n",
      "module.decoder.decoder.0.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.22.conv_list.0.bias\n",
      "module.decoder.decoder.20.last_conv.weight\n",
      "module.decoder.decoder.3.bn1.running_var\n",
      "module.decoder.decoder.5.last_conv.weight\n",
      "module.decoder.decoder.23.conv1.bias\n",
      "module.decoder.decoder.4.bn2.bias\n",
      "module.decoder.decoder.4.bn_list.1.bias\n",
      "module.decoder.decoder.17.bn2.num_batches_tracked\n",
      "module.decoder.decoder.3.conv_list.1.bias\n",
      "module.decoder.decoder.8.conv1.bias\n",
      "module.decoder.decoder.3.bn2.bias\n",
      "module.decoder.decoder.20.bn2.weight\n",
      "module.decoder.decoder.8.bn1.num_batches_tracked\n",
      "module.decoder.decoder.5.bn_list.1.bias\n",
      "module.encoder.bn3.weight\n",
      "module.decoder.decoder.12.bn_list.1.running_mean\n",
      "module.decoder.decoder.22.bn2.running_mean\n",
      "module.encoder.bn4.weight\n",
      "module.decoder.decoder.0.last_conv.bias\n",
      "module.decoder.decoder.23.bn1.running_mean\n",
      "module.decoder.decoder.18.bn1.num_batches_tracked\n",
      "module.decoder.decoder.20.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.22.conv_list.1.bias\n",
      "module.decoder.decoder.10.bn_list.0.weight\n",
      "module.decoder.decoder.3.bn1.bias\n",
      "module.decoder.decoder.22.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.14.bn_list.0.weight\n",
      "module.decoder.decoder.23.bn_list.1.bias\n",
      "module.decoder.decoder.3.bn2.weight\n",
      "module.decoder.decoder.7.conv_list.0.bias\n",
      "module.encoder.bn5.bias\n",
      "module.decoder.decoder.11.bn2.running_mean\n",
      "module.decoder.decoder.4.conv2.bias\n",
      "module.decoder.decoder.10.conv_list.0.bias\n",
      "module.decoder.decoder.1.conv_list.0.bias\n",
      "module.decoder.decoder.2.bn1.running_mean\n",
      "module.decoder.decoder.3.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.0.bn2.weight\n",
      "module.decoder.decoder.13.bn_list.0.running_var\n",
      "module.decoder.decoder.9.bn2.weight\n",
      "module.decoder.decoder.5.bn1.weight\n",
      "module.decoder.decoder.10.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.5.conv_list.0.weight\n",
      "module.decoder.decoder.19.bn_list.1.running_mean\n",
      "module.decoder.decoder.22.bn1.bias\n",
      "module.decoder.decoder.2.conv_list.0.bias\n",
      "module.decoder.decoder.10.conv1.weight\n",
      "module.decoder.decoder.16.conv2.weight\n",
      "module.decoder.decoder.5.bn_list.0.weight\n",
      "module.decoder.decoder.11.conv_list.1.bias\n",
      "module.decoder.decoder.10.conv_list.1.weight\n",
      "module.encoder.bn1.running_var\n",
      "module.decoder.decoder.10.bn_list.0.running_var\n",
      "module.decoder.decoder.9.bn2.bias\n",
      "module.encoder.lin1.weight\n",
      "module.decoder.decoder.9.last_conv.bias\n",
      "module.decoder.decoder.11.conv_list.0.bias\n",
      "module.decoder.decoder.3.conv_list.0.bias\n",
      "module.encoder.conv1.weight\n",
      "module.decoder.decoder.8.last_conv.weight\n",
      "module.decoder.decoder.14.bn2.weight\n",
      "module.decoder.decoder.16.conv_list.0.weight\n",
      "module.decoder.decoder.17.conv_list.1.weight\n",
      "module.decoder.decoder.16.conv_list.0.bias\n",
      "module.decoder.decoder.1.conv1.bias\n",
      "module.decoder.decoder.2.bn_list.1.weight\n",
      "module.decoder.decoder.24.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.1.bn1.num_batches_tracked\n",
      "module.decoder.decoder.14.bn2.bias\n",
      "module.decoder.decoder.3.conv_list.0.weight\n",
      "module.decoder.decoder.18.bn2.running_var\n",
      "module.decoder.decoder.1.bn_list.1.running_mean\n",
      "module.decoder.decoder.23.bn_list.0.running_mean\n",
      "module.decoder.decoder.9.bn_list.1.running_var\n",
      "module.decoder.decoder.19.bn_list.1.bias\n",
      "module.decoder.decoder.24.bn_list.1.weight\n",
      "module.decoder.decoder.19.bn_list.0.weight\n",
      "module.decoder.decoder.22.conv_list.1.weight\n",
      "module.decoder.decoder.20.bn1.bias\n",
      "module.decoder.decoder.9.bn2.num_batches_tracked\n",
      "module.decoder.decoder.16.bn2.bias\n",
      "module.decoder.decoder.12.conv_list.1.bias\n",
      "module.decoder.decoder.15.conv_list.0.weight\n",
      "module.decoder.decoder.18.bn_list.1.running_mean\n",
      "module.decoder.decoder.19.bn1.running_mean\n",
      "module.decoder.decoder.2.conv1.weight\n",
      "module.decoder.decoder.0.conv_list.1.bias\n",
      "module.decoder.decoder.18.bn_list.1.weight\n",
      "module.decoder.decoder.15.bn2.bias\n",
      "module.decoder.decoder.19.conv2.weight\n",
      "module.decoder.decoder.14.bn1.weight\n",
      "module.decoder.decoder.18.bn1.running_var\n",
      "module.decoder.decoder.10.bn2.bias\n",
      "module.decoder.decoder.20.last_conv.bias\n",
      "module.decoder.decoder.24.last_conv.bias\n",
      "module.decoder.decoder.18.last_conv.bias\n",
      "module.decoder.decoder.21.bn_list.0.running_mean\n",
      "module.decoder.decoder.13.bn_list.0.bias\n",
      "module.decoder.decoder.17.conv1.bias\n",
      "module.decoder.decoder.19.bn2.running_var\n",
      "module.decoder.decoder.12.bn1.running_var\n",
      "module.decoder.decoder.16.bn_list.1.running_mean\n",
      "module.encoder.bn1.running_mean\n",
      "module.decoder.decoder.0.conv1.bias\n",
      "module.decoder.decoder.20.conv_list.1.weight\n",
      "module.decoder.decoder.7.last_conv.weight\n",
      "module.decoder.decoder.10.conv2.bias\n",
      "module.decoder.decoder.24.bn1.running_mean\n",
      "module.encoder.bn4.num_batches_tracked\n",
      "module.decoder.decoder.20.bn2.num_batches_tracked\n",
      "module.decoder.decoder.5.bn2.running_var\n",
      "module.decoder.decoder.21.bn2.num_batches_tracked\n",
      "module.encoder.bn3.num_batches_tracked\n",
      "module.decoder.decoder.7.bn_list.0.bias\n",
      "module.decoder.decoder.4.conv_list.1.weight\n",
      "module.decoder.decoder.4.last_conv.bias\n",
      "module.decoder.decoder.5.bn_list.0.running_var\n",
      "module.decoder.decoder.9.bn_list.1.running_mean\n",
      "module.decoder.decoder.0.bn1.running_mean\n",
      "module.decoder.decoder.12.conv1.weight\n",
      "module.decoder.decoder.14.bn_list.1.weight\n",
      "module.decoder.decoder.16.conv_list.1.weight\n",
      "module.decoder.decoder.1.bn_list.1.weight\n",
      "module.decoder.decoder.14.conv2.bias\n",
      "module.decoder.decoder.0.conv1.weight\n",
      "module.decoder.decoder.3.last_conv.bias\n",
      "module.decoder.decoder.13.conv_list.1.bias\n",
      "module.decoder.decoder.10.bn1.running_var\n",
      "module.decoder.decoder.23.bn2.running_var\n",
      "module.decoder.decoder.7.bn1.bias\n",
      "module.decoder.decoder.13.conv_list.0.bias\n",
      "module.decoder.decoder.22.bn1.running_mean\n",
      "module.encoder.bn1.num_batches_tracked\n",
      "module.decoder.decoder.10.bn1.bias\n",
      "module.decoder.decoder.0.last_conv.weight\n",
      "module.decoder.decoder.1.bn1.bias\n",
      "module.decoder.decoder.9.conv2.bias\n",
      "module.decoder.decoder.0.bn1.num_batches_tracked\n",
      "module.decoder.decoder.7.bn_list.0.running_mean\n",
      "module.decoder.decoder.1.bn2.running_mean\n",
      "module.decoder.decoder.13.conv1.bias\n",
      "module.decoder.decoder.0.bn_list.1.weight\n",
      "module.decoder.decoder.2.bn_list.0.running_var\n",
      "module.decoder.decoder.3.bn_list.0.running_var\n",
      "module.decoder.decoder.12.bn_list.1.weight\n",
      "module.decoder.decoder.20.conv_list.0.bias\n",
      "module.decoder.decoder.21.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.19.bn1.weight\n",
      "module.decoder.decoder.16.last_conv.weight\n",
      "module.decoder.decoder.9.bn1.running_mean\n",
      "module.decoder.decoder.7.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.7.bn2.weight\n",
      "module.decoder.decoder.13.bn2.running_mean\n",
      "module.decoder.decoder.4.bn2.running_mean\n",
      "module.decoder.decoder.10.bn2.num_batches_tracked\n",
      "module.encoder.bn3.bias\n",
      "module.decoder.decoder.21.bn1.running_var\n",
      "module.decoder.decoder.9.bn_list.0.running_var\n",
      "module.decoder.decoder.13.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.18.conv2.weight\n",
      "module.decoder.decoder.21.bn_list.0.weight\n",
      "module.decoder.decoder.8.bn_list.0.running_mean\n",
      "module.decoder.decoder.6.bn1.running_var\n",
      "module.decoder.decoder.17.bn2.running_var\n",
      "module.decoder.decoder.18.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.23.bn2.num_batches_tracked\n",
      "module.decoder.decoder.16.bn_list.0.weight\n",
      "module.encoder.bn2.num_batches_tracked\n",
      "module.decoder.decoder.17.bn_list.0.running_var\n",
      "module.decoder.decoder.1.bn2.bias\n",
      "module.decoder.decoder.22.last_conv.weight\n",
      "module.decoder.decoder.9.bn1.num_batches_tracked\n",
      "module.decoder.decoder.8.bn_list.1.running_var\n",
      "module.decoder.decoder.14.bn2.running_mean\n",
      "module.decoder.decoder.11.bn2.num_batches_tracked\n",
      "module.decoder.decoder.3.bn1.num_batches_tracked\n",
      "module.decoder.decoder.14.conv_list.1.bias\n",
      "module.decoder.decoder.14.bn_list.0.running_var\n",
      "module.decoder.decoder.7.bn_list.0.weight\n",
      "module.decoder.decoder.6.conv1.weight\n",
      "module.decoder.decoder.14.bn_list.0.bias\n",
      "module.decoder.decoder.21.bn_list.1.running_var\n",
      "module.decoder.decoder.3.bn1.weight\n",
      "module.decoder.decoder.13.bn2.num_batches_tracked\n",
      "module.decoder.decoder.15.bn_list.1.running_var\n",
      "module.decoder.decoder.1.bn1.running_var\n",
      "module.decoder.decoder.8.bn_list.0.running_var\n",
      "module.decoder.decoder.19.last_conv.bias\n",
      "module.decoder.decoder.14.conv2.weight\n",
      "module.decoder.decoder.17.bn_list.0.running_mean\n",
      "module.encoder.bn4.running_mean\n",
      "module.decoder.decoder.13.bn1.running_mean\n",
      "module.decoder.decoder.13.bn2.bias\n",
      "module.decoder.decoder.18.conv_list.1.bias\n",
      "module.decoder.decoder.17.bn_list.0.bias\n",
      "module.decoder.decoder.5.conv1.weight\n",
      "module.decoder.decoder.4.bn2.weight\n",
      "module.decoder.decoder.6.bn_list.0.weight\n",
      "module.decoder.decoder.0.bn_list.1.running_mean\n",
      "module.encoder.bn1.weight\n",
      "module.decoder.decoder.8.bn2.running_var\n",
      "module.decoder.decoder.23.bn_list.0.running_var\n",
      "module.decoder.decoder.22.bn2.num_batches_tracked\n",
      "module.decoder.decoder.7.conv_list.1.bias\n",
      "module.decoder.decoder.9.conv1.weight\n",
      "module.encoder.lin2.weight\n",
      "module.decoder.decoder.6.conv_list.0.bias\n",
      "module.decoder.decoder.21.conv_list.1.weight\n",
      "module.decoder.decoder.4.bn_list.0.running_var\n",
      "module.decoder.decoder.21.bn_list.0.running_var\n",
      "module.decoder.decoder.19.bn_list.1.running_var\n",
      "module.decoder.decoder.0.conv_list.1.weight\n",
      "module.decoder.decoder.23.bn1.running_var\n",
      "module.decoder.decoder.8.conv_list.0.bias\n",
      "module.decoder.decoder.10.conv2.weight\n",
      "module.decoder.decoder.15.last_conv.weight\n",
      "module.decoder.decoder.21.bn_list.1.running_mean\n",
      "module.decoder.decoder.10.bn1.running_mean\n",
      "module.decoder.decoder.15.conv_list.1.weight\n",
      "module.decoder.decoder.15.bn_list.0.weight\n",
      "module.decoder.decoder.0.bn_list.1.bias\n",
      "module.decoder.decoder.3.bn_list.0.bias\n",
      "module.decoder.decoder.4.bn1.running_mean\n",
      "module.decoder.decoder.9.bn2.running_mean\n",
      "module.decoder.decoder.21.last_conv.weight\n",
      "module.decoder.decoder.23.bn2.running_mean\n",
      "module.decoder.decoder.0.conv2.bias\n",
      "module.decoder.decoder.5.conv_list.1.bias\n",
      "module.decoder.decoder.19.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.4.conv1.bias\n",
      "module.decoder.decoder.22.bn_list.1.bias\n",
      "module.decoder.decoder.0.bn_list.0.running_mean\n",
      "module.decoder.decoder.14.bn1.running_var\n",
      "module.decoder.decoder.21.conv_list.1.bias\n",
      "module.decoder.decoder.2.conv_list.1.bias\n",
      "module.decoder.decoder.6.conv2.bias\n",
      "module.decoder.decoder.22.bn_list.0.running_mean\n",
      "module.decoder.decoder.7.bn_list.1.running_mean\n",
      "module.decoder.decoder.10.last_conv.bias\n",
      "module.decoder.decoder.2.bn1.running_var\n",
      "module.decoder.decoder.4.bn1.weight\n",
      "module.decoder.decoder.15.bn_list.0.running_mean\n",
      "module.decoder.decoder.21.bn1.bias\n",
      "module.decoder.decoder.4.bn_list.1.weight\n",
      "module.decoder.decoder.6.bn1.num_batches_tracked\n",
      "module.decoder.decoder.5.conv1.bias\n",
      "module.decoder.decoder.1.bn_list.0.running_mean\n",
      "module.decoder.decoder.21.conv2.weight\n",
      "module.decoder.decoder.8.bn_list.0.weight\n",
      "module.decoder.decoder.15.bn2.running_mean\n",
      "module.decoder.decoder.7.conv_list.1.weight\n",
      "module.decoder.decoder.11.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.20.bn_list.0.running_var\n",
      "module.decoder.decoder.11.bn1.running_var\n",
      "module.decoder.decoder.15.bn_list.1.running_mean\n",
      "module.decoder.decoder.4.bn2.running_var\n",
      "module.decoder.decoder.16.conv_list.1.bias\n",
      "module.decoder.decoder.8.conv_list.1.bias\n",
      "module.decoder.decoder.4.bn2.num_batches_tracked\n",
      "module.decoder.decoder.19.bn_list.0.running_mean\n",
      "module.decoder.decoder.12.bn_list.0.running_mean\n",
      "module.decoder.decoder.17.bn_list.1.weight\n",
      "module.decoder.decoder.19.conv_list.1.weight\n",
      "module.decoder.decoder.17.bn_list.1.bias\n",
      "module.decoder.decoder.17.bn_list.0.weight\n",
      "module.decoder.decoder.7.bn2.bias\n",
      "module.decoder.decoder.12.conv_list.0.bias\n",
      "module.decoder.decoder.2.bn_list.0.running_mean\n",
      "module.decoder.decoder.3.bn2.num_batches_tracked\n",
      "module.decoder.decoder.16.bn_list.0.running_mean\n",
      "module.decoder.decoder.11.bn_list.1.bias\n",
      "module.decoder.decoder.7.bn1.running_var\n",
      "module.decoder.decoder.1.bn1.weight\n",
      "module.decoder.decoder.12.bn_list.0.weight\n",
      "module.decoder.decoder.8.conv_list.1.weight\n",
      "module.decoder.decoder.8.bn2.bias\n",
      "module.decoder.decoder.15.bn1.running_mean\n",
      "module.decoder.decoder.9.bn1.running_var\n",
      "module.decoder.decoder.12.bn2.bias\n",
      "module.decoder.decoder.20.conv1.bias\n",
      "module.decoder.decoder.24.bn1.bias\n",
      "module.decoder.decoder.12.last_conv.weight\n",
      "module.decoder.decoder.18.bn2.weight\n",
      "module.decoder.decoder.22.bn_list.0.weight\n",
      "module.decoder.decoder.16.bn1.bias\n",
      "module.decoder.decoder.23.bn_list.0.weight\n",
      "module.decoder.decoder.6.bn2.num_batches_tracked\n",
      "module.encoder.bn4.bias\n",
      "module.decoder.decoder.1.conv2.weight\n",
      "module.decoder.decoder.1.bn_list.1.running_var\n",
      "module.decoder.decoder.2.last_conv.weight\n",
      "module.decoder.decoder.8.conv2.bias\n",
      "module.decoder.decoder.8.bn2.running_mean\n",
      "module.decoder.decoder.17.bn1.running_mean\n",
      "module.decoder.decoder.6.last_conv.weight\n",
      "module.decoder.decoder.0.bn2.bias\n",
      "module.decoder.decoder.20.bn1.running_var\n",
      "module.decoder.decoder.1.bn2.weight\n",
      "module.encoder.bn5.num_batches_tracked\n",
      "module.decoder.decoder.7.bn2.running_mean\n",
      "module.decoder.decoder.14.bn_list.0.running_mean\n",
      "module.decoder.decoder.11.conv2.weight\n",
      "module.decoder.decoder.3.conv1.weight\n",
      "module.decoder.decoder.4.conv_list.1.bias\n",
      "module.decoder.decoder.9.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.20.bn_list.0.weight\n",
      "module.decoder.decoder.5.conv2.bias\n",
      "module.decoder.decoder.0.bn1.weight\n",
      "module.decoder.decoder.5.conv_list.0.bias\n",
      "module.encoder.bn2.running_mean\n",
      "module.decoder.decoder.3.conv_list.1.weight\n",
      "module.decoder.decoder.13.conv2.bias\n",
      "module.decoder.decoder.10.bn_list.1.bias\n",
      "module.decoder.decoder.17.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.23.conv2.weight\n",
      "module.decoder.decoder.12.bn1.running_mean\n",
      "module.decoder.decoder.1.bn2.running_var\n",
      "module.decoder.decoder.11.bn_list.0.weight\n",
      "module.decoder.decoder.0.conv_list.0.bias\n",
      "module.decoder.decoder.7.bn_list.1.running_var\n",
      "module.decoder.decoder.13.conv_list.0.weight\n",
      "module.decoder.decoder.16.bn1.running_var\n",
      "module.decoder.decoder.18.conv_list.1.weight\n",
      "module.decoder.decoder.6.bn2.running_var\n",
      "module.decoder.decoder.24.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.9.conv_list.1.bias\n",
      "module.decoder.decoder.22.last_conv.bias\n",
      "module.decoder.decoder.14.last_conv.weight\n",
      "module.decoder.decoder.11.conv_list.1.weight\n",
      "module.decoder.decoder.16.bn1.num_batches_tracked\n",
      "module.decoder.decoder.14.bn2.running_var\n",
      "module.decoder.decoder.12.bn1.num_batches_tracked\n",
      "module.decoder.decoder.23.conv_list.0.weight\n",
      "module.decoder.decoder.1.conv_list.1.bias\n",
      "module.encoder.bn3.running_var\n",
      "module.decoder.decoder.18.conv2.bias\n",
      "module.decoder.decoder.15.bn2.running_var\n",
      "module.decoder.decoder.18.bn1.bias\n",
      "module.decoder.decoder.20.bn_list.1.running_var\n",
      "module.decoder.decoder.21.bn2.weight\n",
      "module.decoder.decoder.3.bn_list.1.running_var\n",
      "module.decoder.decoder.19.bn1.num_batches_tracked\n",
      "module.decoder.decoder.8.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.6.conv_list.1.weight\n",
      "module.decoder.decoder.23.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.1.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.3.bn_list.0.weight\n",
      "module.decoder.decoder.14.bn2.num_batches_tracked\n",
      "module.decoder.decoder.20.bn2.bias\n",
      "module.decoder.decoder.20.bn1.num_batches_tracked\n",
      "module.decoder.decoder.24.conv_list.0.bias\n",
      "module.decoder.decoder.11.bn2.bias\n",
      "module.decoder.decoder.9.conv_list.0.bias\n",
      "module.decoder.decoder.23.bn_list.0.bias\n",
      "module.decoder.decoder.21.bn1.num_batches_tracked\n",
      "module.decoder.decoder.15.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.2.bn_list.1.running_mean\n",
      "module.decoder.decoder.12.bn_list.0.bias\n",
      "module.decoder.decoder.13.bn_list.0.running_mean\n",
      "module.decoder.decoder.13.bn2.running_var\n",
      "module.decoder.decoder.11.conv2.bias\n",
      "module.decoder.decoder.18.conv_list.0.weight\n",
      "module.decoder.decoder.14.bn_list.1.num_batches_tracked\n",
      "module.encoder.conv3.weight\n",
      "module.decoder.decoder.19.conv1.bias\n",
      "module.decoder.decoder.24.bn2.bias\n",
      "module.decoder.decoder.15.conv_list.0.bias\n",
      "module.decoder.decoder.0.conv2.weight\n",
      "module.decoder.decoder.24.conv2.weight\n",
      "module.decoder.decoder.5.bn2.num_batches_tracked\n",
      "module.decoder.decoder.6.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.17.bn2.bias\n",
      "module.decoder.decoder.4.bn1.num_batches_tracked\n",
      "module.decoder.decoder.4.bn_list.1.running_var\n",
      "module.decoder.decoder.19.last_conv.weight\n",
      "module.encoder.bn2.bias\n",
      "module.decoder.decoder.5.bn2.weight\n",
      "module.decoder.decoder.4.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.14.conv1.bias\n",
      "module.decoder.decoder.19.bn2.bias\n",
      "module.decoder.decoder.11.bn_list.1.running_mean\n",
      "module.decoder.decoder.22.bn1.weight\n",
      "module.decoder.decoder.12.bn2.running_mean\n",
      "module.decoder.decoder.7.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.18.bn_list.0.weight\n",
      "module.decoder.decoder.5.conv_list.1.weight\n",
      "module.decoder.decoder.7.bn1.running_mean\n",
      "module.decoder.decoder.13.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.18.bn1.running_mean\n",
      "module.decoder.decoder.6.conv1.bias\n",
      "module.decoder.decoder.19.bn2.weight\n",
      "module.decoder.decoder.17.bn_list.1.running_var\n",
      "module.decoder.decoder.13.last_conv.bias\n",
      "module.decoder.decoder.19.conv_list.1.bias\n",
      "module.decoder.decoder.1.conv_list.1.weight\n",
      "module.decoder.decoder.18.bn2.num_batches_tracked\n",
      "module.decoder.decoder.11.bn_list.1.weight\n",
      "module.decoder.decoder.19.bn1.bias\n",
      "module.decoder.decoder.20.bn_list.1.weight\n",
      "module.decoder.decoder.14.conv_list.1.weight\n",
      "module.decoder.decoder.19.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.15.bn2.weight\n",
      "module.decoder.decoder.10.bn2.running_mean\n",
      "module.decoder.decoder.13.bn_list.1.bias\n",
      "module.decoder.decoder.23.bn2.bias\n",
      "module.decoder.decoder.17.bn_list.1.running_mean\n",
      "module.decoder.decoder.16.bn1.running_mean\n",
      "module.decoder.decoder.17.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.21.bn_list.1.weight\n",
      "module.decoder.decoder.21.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.18.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.0.bn2.running_mean\n",
      "module.decoder.decoder.4.bn1.running_var\n",
      "module.decoder.decoder.17.last_conv.bias\n",
      "module.decoder.decoder.10.bn_list.0.running_mean\n",
      "module.decoder.decoder.24.bn1.weight\n",
      "module.decoder.decoder.2.bn_list.1.running_var\n",
      "module.decoder.decoder.24.bn_list.0.running_var\n",
      "module.decoder.decoder.13.bn2.weight\n",
      "module.decoder.decoder.20.bn2.running_var\n",
      "module.decoder.decoder.24.bn2.weight\n",
      "module.decoder.decoder.7.conv2.bias\n",
      "module.decoder.decoder.12.bn2.weight\n",
      "module.decoder.decoder.6.bn2.weight\n",
      "module.decoder.decoder.10.bn1.num_batches_tracked\n",
      "module.decoder.decoder.13.bn1.bias\n",
      "module.decoder.decoder.8.bn1.bias\n",
      "module.decoder.decoder.17.bn1.bias\n",
      "module.decoder.decoder.16.bn2.running_var\n",
      "module.encoder.lin2.bias\n",
      "module.decoder.decoder.11.bn1.weight\n",
      "module.decoder.decoder.6.bn_list.1.running_var\n",
      "module.decoder.decoder.0.bn_list.1.running_var\n",
      "module.decoder.decoder.1.conv1.weight\n",
      "module.decoder.decoder.17.conv2.weight\n",
      "module.decoder.decoder.23.conv_list.1.bias\n",
      "module.decoder.decoder.8.bn1.weight\n",
      "module.decoder.decoder.18.conv_list.0.bias\n",
      "module.decoder.decoder.12.bn1.weight\n",
      "module.decoder.decoder.13.last_conv.weight\n",
      "module.decoder.decoder.22.conv1.weight\n",
      "module.decoder.decoder.22.conv1.bias\n",
      "module.decoder.decoder.19.bn_list.1.weight\n",
      "module.decoder.decoder.23.last_conv.weight\n",
      "module.decoder.decoder.21.conv_list.0.weight\n",
      "module.decoder.decoder.22.bn_list.0.running_var\n",
      "module.decoder.decoder.18.bn_list.0.bias\n",
      "module.decoder.decoder.2.bn_list.1.bias\n",
      "module.decoder.decoder.24.bn1.num_batches_tracked\n",
      "module.decoder.decoder.24.bn_list.0.running_mean\n",
      "module.decoder.decoder.14.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.22.bn2.weight\n",
      "module.decoder.decoder.8.bn_list.1.weight\n",
      "module.decoder.decoder.20.bn_list.0.running_mean\n",
      "module.decoder.decoder.0.bn_list.0.bias\n",
      "module.decoder.decoder.12.conv2.weight\n",
      "module.decoder.decoder.21.bn1.running_mean\n",
      "module.decoder.decoder.24.conv1.bias\n",
      "module.decoder.decoder.5.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.16.conv1.bias\n",
      "module.decoder.decoder.5.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.4.bn_list.0.running_mean\n",
      "module.decoder.decoder.9.conv_list.0.weight\n",
      "module.decoder.decoder.15.conv_list.1.bias\n",
      "module.decoder.decoder.3.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.9.bn_list.1.bias\n",
      "module.decoder.decoder.18.last_conv.weight\n",
      "module.decoder.decoder.21.conv1.weight\n",
      "module.decoder.decoder.11.bn1.bias\n",
      "module.decoder.decoder.6.bn_list.1.bias\n",
      "module.decoder.decoder.17.conv1.weight\n",
      "module.decoder.decoder.24.conv2.bias\n",
      "module.decoder.decoder.16.conv2.bias\n",
      "module.decoder.decoder.3.bn_list.0.running_mean\n",
      "module.decoder.decoder.1.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.14.bn_list.1.running_mean\n",
      "module.decoder.decoder.22.bn2.running_var\n",
      "module.decoder.decoder.16.bn1.weight\n",
      "module.decoder.decoder.6.bn_list.0.running_mean\n",
      "module.decoder.decoder.7.conv2.weight\n",
      "module.decoder.decoder.17.conv_list.0.weight\n",
      "module.decoder.decoder.23.bn2.weight\n",
      "module.decoder.decoder.8.bn2.weight\n",
      "module.decoder.decoder.15.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.2.bn_list.0.bias\n",
      "module.decoder.decoder.20.bn1.weight\n",
      "module.decoder.decoder.9.last_conv.weight\n",
      "module.decoder.decoder.7.last_conv.bias\n",
      "module.decoder.decoder.12.conv2.bias\n",
      "module.decoder.decoder.13.bn_list.1.running_mean\n",
      "module.decoder.decoder.12.conv1.bias\n",
      "module.decoder.decoder.8.conv1.weight\n",
      "module.decoder.decoder.4.conv2.weight\n",
      "module.decoder.decoder.11.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.21.conv_list.0.bias\n",
      "module.decoder.decoder.14.bn1.running_mean\n",
      "module.decoder.decoder.23.bn1.weight\n",
      "module.decoder.decoder.7.conv_list.0.weight\n",
      "module.decoder.decoder.18.bn_list.0.running_var\n",
      "module.decoder.decoder.5.bn_list.0.bias\n",
      "module.decoder.decoder.19.conv1.weight\n",
      "module.decoder.decoder.24.conv1.weight\n",
      "module.encoder.bn5.running_mean\n",
      "module.decoder.decoder.7.bn_list.1.weight\n",
      "module.decoder.decoder.3.conv1.bias\n",
      "module.decoder.decoder.15.bn_list.1.weight\n",
      "module.decoder.decoder.4.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.6.bn_list.1.running_mean\n",
      "module.decoder.decoder.19.bn2.running_mean\n",
      "module.decoder.decoder.20.bn2.running_mean\n",
      "module.decoder.decoder.13.bn_list.0.weight\n",
      "module.decoder.decoder.17.bn2.running_mean\n",
      "module.decoder.decoder.14.last_conv.bias\n",
      "module.decoder.decoder.24.bn2.running_mean\n",
      "module.decoder.decoder.22.bn2.bias\n",
      "module.decoder.decoder.1.bn_list.0.running_var\n",
      "module.decoder.decoder.16.bn2.num_batches_tracked\n",
      "module.decoder.decoder.24.bn_list.0.weight\n",
      "module.decoder.decoder.7.conv1.bias\n",
      "module.decoder.decoder.15.conv1.bias\n",
      "module.decoder.decoder.2.bn1.num_batches_tracked\n",
      "module.decoder.decoder.6.bn2.running_mean\n",
      "module.decoder.decoder.19.bn_list.0.running_var\n",
      "module.decoder.decoder.15.bn_list.0.running_var\n",
      "module.decoder.decoder.4.conv_list.0.bias\n",
      "module.decoder.decoder.6.bn_list.0.bias\n",
      "module.decoder.decoder.20.bn_list.1.running_mean\n",
      "module.decoder.decoder.5.bn1.num_batches_tracked\n",
      "module.decoder.decoder.0.bn_list.0.running_var\n",
      "module.decoder.decoder.17.last_conv.weight\n",
      "module.decoder.decoder.10.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.5.bn1.running_var\n",
      "module.decoder.decoder.13.bn1.weight\n",
      "module.decoder.decoder.23.conv_list.0.bias\n",
      "module.decoder.decoder.14.conv1.weight\n",
      "module.decoder.decoder.5.bn1.running_mean\n",
      "module.decoder.decoder.21.bn2.bias\n",
      "module.decoder.decoder.22.conv2.bias\n",
      "module.decoder.decoder.11.bn2.running_var\n",
      "module.decoder.decoder.1.bn2.num_batches_tracked\n",
      "module.decoder.decoder.9.bn1.bias\n",
      "module.decoder.decoder.0.bn2.running_var\n",
      "module.decoder.decoder.10.bn1.weight\n",
      "module.decoder.decoder.24.bn_list.1.bias\n",
      "module.decoder.decoder.15.bn1.running_var\n",
      "module.decoder.decoder.2.conv_list.0.weight\n",
      "module.decoder.decoder.16.conv1.weight\n",
      "module.decoder.decoder.20.conv_list.1.bias\n",
      "module.decoder.decoder.0.bn1.bias\n",
      "module.decoder.decoder.2.bn2.num_batches_tracked\n",
      "module.decoder.decoder.14.bn_list.1.bias\n",
      "module.decoder.decoder.24.conv_list.1.bias\n",
      "module.decoder.decoder.22.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.10.conv_list.1.bias\n",
      "module.decoder.decoder.3.bn_list.1.running_mean\n",
      "module.decoder.decoder.16.bn2.running_mean\n",
      "module.decoder.decoder.12.bn_list.1.num_batches_tracked\n",
      "module.decoder.decoder.10.conv1.bias\n",
      "module.decoder.decoder.0.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.19.bn_list.0.bias\n",
      "module.decoder.decoder.7.conv1.weight\n",
      "module.decoder.decoder.9.bn_list.0.weight\n",
      "module.decoder.decoder.23.bn_list.1.running_var\n",
      "module.decoder.decoder.14.bn1.bias\n",
      "module.decoder.decoder.23.conv2.bias\n",
      "module.decoder.decoder.15.bn1.num_batches_tracked\n",
      "module.decoder.decoder.6.conv_list.0.weight\n",
      "module.decoder.decoder.2.bn2.bias\n",
      "module.decoder.decoder.5.bn_list.1.running_var\n",
      "module.decoder.decoder.10.bn_list.1.weight\n",
      "module.decoder.decoder.13.conv2.weight\n",
      "module.decoder.decoder.6.bn1.running_mean\n",
      "module.decoder.decoder.24.bn2.running_var\n",
      "module.decoder.decoder.20.bn_list.1.bias\n",
      "module.decoder.decoder.5.last_conv.bias\n",
      "module.decoder.decoder.5.bn_list.1.weight\n",
      "module.decoder.decoder.13.conv_list.1.weight\n",
      "module.encoder.bn2.running_var\n",
      "module.decoder.decoder.4.conv1.weight\n",
      "module.decoder.decoder.10.bn_list.1.running_var\n",
      "module.decoder.decoder.11.last_conv.weight\n",
      "module.decoder.decoder.8.conv_list.0.weight\n",
      "module.decoder.decoder.0.bn1.running_var\n",
      "module.decoder.decoder.18.bn2.running_mean\n",
      "module.decoder.decoder.12.bn_list.1.running_var\n",
      "module.encoder.conv2.bias\n",
      "module.decoder.decoder.9.bn_list.1.weight\n",
      "module.decoder.decoder.2.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.15.bn2.num_batches_tracked\n",
      "module.decoder.decoder.10.bn_list.1.running_mean\n",
      "module.decoder.decoder.19.bn2.num_batches_tracked\n",
      "module.decoder.decoder.21.conv1.bias\n",
      "module.decoder.decoder.15.last_conv.bias\n",
      "module.decoder.decoder.2.bn2.running_mean\n",
      "module.decoder.decoder.8.bn_list.0.bias\n",
      "module.decoder.decoder.0.bn2.num_batches_tracked\n",
      "module.decoder.decoder.20.conv2.bias\n",
      "module.encoder.bn2.weight\n",
      "module.encoder.bn5.running_var\n",
      "module.encoder.bn3.running_mean\n",
      "module.decoder.decoder.4.last_conv.weight\n",
      "module.decoder.decoder.9.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.16.bn2.weight\n",
      "module.decoder.decoder.24.last_conv.weight\n",
      "module.decoder.decoder.16.bn_list.0.running_var\n",
      "module.decoder.decoder.3.conv2.weight\n",
      "module.decoder.decoder.1.bn_list.1.bias\n",
      "module.decoder.decoder.14.conv_list.0.bias\n",
      "module.decoder.decoder.16.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.12.conv_list.1.weight\n",
      "module.decoder.decoder.23.bn_list.1.weight\n",
      "module.decoder.decoder.2.conv1.bias\n",
      "module.decoder.decoder.6.bn_list.0.num_batches_tracked\n",
      "module.decoder.decoder.5.bn1.bias\n",
      "module.decoder.decoder.17.conv_list.0.bias\n",
      "module.decoder.decoder.1.last_conv.weight\n",
      "module.decoder.decoder.15.conv1.weight\n",
      "module.decoder.decoder.5.bn2.bias\n",
      "module.decoder.decoder.9.conv_list.1.weight\n",
      "module.decoder.decoder.17.conv2.bias\n",
      "module.decoder.decoder.0.bn_list.0.weight\n",
      "module.decoder.decoder.17.bn1.num_batches_tracked\n",
      "module.decoder.decoder.22.bn_list.0.bias\n",
      "module.decoder.decoder.16.bn_list.1.running_var\n",
      "module.decoder.decoder.20.bn1.running_mean\n",
      "两个.pth文件存在差异:\n",
      "模型状态字典存在差异\n"
     ]
    }
   ],
   "source": [
    "\n",
    "import torch\n",
    "\n",
    "def compare_models(file1, file2):\n",
    "    model1 = torch.load(file1)\n",
    "    model2 = torch.load(file2)\n",
    "    \n",
    "    \n",
    "    \n",
    "    # 比较模型状态字典\n",
    "    state_dict_diff = compare_state_dicts(model1, model2)\n",
    "    \n",
    "    if state_dict_diff:\n",
    "        print(\"两个.pth文件存在差异:\")\n",
    "        if state_dict_diff:\n",
    "            print(\"模型状态字典存在差异\")\n",
    "    else:\n",
    "        print(\"两个.pth文件完全相同\")\n",
    "\n",
    "def compare_state_dicts(dict1, dict2):\n",
    "    # 比较状态字典的键\n",
    "    keys1 = set(dict1.keys())\n",
    "    keys2 = set(dict2.keys())\n",
    "    \n",
    "    # if keys1 != keys2:\n",
    "    #     return True\n",
    "    \n",
    "    # 比较每个键对应的值\n",
    "    flog = False\n",
    "    for key in keys1:\n",
    "        if dict2.get(key) is None or torch.any(dict1[key] != dict2[key]) :\n",
    "            print(key)\n",
    "            flog = True\n",
    "\n",
    "    return flog\n",
    "\n",
    "# 调用示例\n",
    "file1 = \"/home/DISCOVER_summer2022/xusc/exp/AtlasNet/logs/atlasnet_autoencoder_25_squares/network.pth\"\n",
    "file2 = \"/home/DISCOVER_summer2022/xusc/exp/AtlasNet/training/trained_models/atlasnet_autoencoder_25_squares/network.pth\"\n",
    "compare_models(file1, file2)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "atlasnetV2",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}
